from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

from .base import Base
from .task_entity import CodeReviewTask
from .task_entity import TaskStatus
from .task_entity import Priority
from .user_entity import UserRole, User
from .login_record_entity import LoginRecord

engine = create_engine('sqlite:///:memory:', echo=True)
Session = sessionmaker(bind=engine)


def create_all_tables():
    Base.metadata.create_all(engine)
